package com.ys.ffms.utils;

import sun.misc.BASE64Decoder;
import sun.misc.BASE64Encoder;

import javax.crypto.Cipher;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;


/**
 * @Classname ZzSecurityHelper
 * @Description TODO
 * @Date 2019/6/24 16:50
 * @Created by whd
 */
public class AESUtil {
    /*
     * 加密用的Key 可以用26个字母和数字组成 使用AES-128-CBC加密模式，key需要为16位。
     */
    private static final String key = "hj7x89H$yuBI0456";
    private static final String iv = "NIfb&95GUY86Gfgh";



    public static void main(String[] args)throws Exception {
        String s = AESUtil.encryptAES("qewqeweqewqeqeww1");//加密
        System.out.println(s);
//
        String s1 = AESUtil.decryptAES(s);//解密
        System.out.println(s1);

    }

    /**
     * @param data 明文
//     * @param key  密钥，长度16
//     * @param iv   偏移量，长度16
     * @return 密文
     * @author miracle.qu
     * @Description AES算法加密明文
     */
    public static String encryptAES(String data) throws Exception {
        try {
            //获取加密实例
            Cipher cipher = Cipher.getInstance("AES/CTR/PKCS5Padding");
            //秘钥规格
            SecretKeySpec keyspec = new SecretKeySpec(key.getBytes(), "AES");
            //iv规格
            IvParameterSpec ivspec = new IvParameterSpec(iv.getBytes());  // CBC模式，需要一个向量iv，可增加加密算法的强度
            //初始化密钥
            cipher.init(Cipher.ENCRYPT_MODE, keyspec, ivspec);


            byte[] dataBytes = data.getBytes();
            //最后加密
            byte[] encrypted = cipher.doFinal(dataBytes);
            String encode = new BASE64Encoder().encode(encrypted);
            return encode; // BASE64做转码。
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

    /**
     * @param data 密文
//     * @param key  密钥，长度16
//     * @param iv   偏移量，长度16
     * @return 明文
     * @author miracle.qu
     * @Description AES算法解密密文
     */
    public static String decryptAES(String data) throws Exception {
        try {
            byte[] encrypted1 =new BASE64Decoder().decodeBuffer(data);//先用base64解密

            Cipher cipher = Cipher.getInstance("AES/CTR/PKCS5Padding");
            SecretKeySpec keyspec = new SecretKeySpec(key.getBytes(), "AES");
            IvParameterSpec ivspec = new IvParameterSpec(iv.getBytes());

            cipher.init(Cipher.DECRYPT_MODE, keyspec, ivspec);

            byte[] original = cipher.doFinal(encrypted1);
            String originalString = new String(original);
            return originalString.trim();
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

}